import os
api_key = os.environ.get("ARK_API_KEY")

TEMPERATURE = 0.8
#
# LLM_MODELS = ["chatglm3-6b", "zhipu-api"]

LLM_MODELS = ["zhipu-api"]

MODEL_PATH = {
    # 这里定义 本机服务器上存储的大模型权重存储路径
    "local_model": {
        "chatglm3-6b": None,

        # 可扩展其他的开源大模型

    },
    
    # 这里定义 本机服务器上存储的Embedding模型权重存储路径
    "embed_model": {
        "bge-large-zh-v1.5": None,

        # 可扩展其他的Embedding模型
    },
}


ONLINE_LLM_MODEL = {


    "zhipu-api": {
        "model_name": "deepseek-v3-250324",
        "api_base_url": "https://ark.cn-beijing.volces.com/api/v3",
        "api_key": api_key,
        "openai_proxy": "",
    },

    "embed_model": {
        "model_name": "doubao-embedding-text-240715",
        "api_base_url": "https://ark.cn-beijing.volces.com/api/v3",
        "api_key": api_key,
        "openai_proxy": "",
    },

    # 可扩展其他的模型在线模型
    
}




